package com.hbmy.workqueue;

import java.io.IOException;
import java.util.concurrent.TimeoutException;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;
import com.rabbitmq.client.Consumer;
import com.rabbitmq.client.DefaultConsumer;
import com.rabbitmq.client.Envelope;
import com.rabbitmq.client.AMQP.BasicProperties;

public class Worker
{
	private final static String TASK_QUEUE_NAME = "task_queue";

	public static void main(String[] args)
	{
		ConnectionFactory factory = new ConnectionFactory();
		factory.setHost("localhost");
		factory.setPort(5682);

		try
		{
			Connection connection = factory.newConnection();
			Channel channel = connection.createChannel();
			channel.queueDeclare(TASK_QUEUE_NAME, true, false, false, null);

			System.out.println(" [*] Waiting for messages. To exit press CTRL+C");
			channel.basicQos(1);

			Consumer consumer = new DefaultConsumer(channel)
			{
				@Override
				public void handleDelivery(String consumerTag, Envelope envelope, BasicProperties properties,
						byte[] body) throws IOException
				{
					String message = new String(body, "UTF-8");
					System.out.println(" [x] Received '" + message + "'");
					try
					{
						doWork(message);
					} finally
					{
						// Message acknowledgment
						System.out.println(" [x] Done");
						System.out.println(envelope.getDeliveryTag());
						channel.basicAck(envelope.getDeliveryTag(), false);
					}
				}
			};
			channel.basicConsume(TASK_QUEUE_NAME, false, consumer);

		} catch (IOException e)
		{
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (TimeoutException e)
		{
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

	private static void doWork(String task)
	{
		for (char ch : task.toCharArray())
		{
			if (ch == '.')
			{
				try
				{
					Thread.sleep(1000L);
				} catch (InterruptedException e)
				{
					Thread.currentThread().interrupt();
				}
			}
		}
	}
}
